
import requests
import json

# 定义API的URL
url = "http://localhost:11434/api/chat"

# 初始化对话历史记录
messages = []

def chat_with_model(user_input):
    # 将用户输入添加到对话历史
    messages.append({"role": "user", "content": user_input})

    # 构建请求数据
    data = {
        "model": "deepseek-r1:1.5b",
        "messages": messages,
        "stream": False  # 设置为False以获取完整的响应
    }

    # 发送POST请求
    response = requests.post(url, json=data)

    # 检查响应状态码
    if response.status_code == 200:
        result = response.json()
        # 提取模型的回复内容
        model_reply = result.get("message", {}).get("content", "")
        if model_reply:
            # 将模型的回复添加到对话历史
            messages.append({"role": "assistant", "content": model_reply})
            return model_reply
        else:
            return "模型未返回有效的回复。"
    else:
        return f"请求失败，状态码: {response.status_code}, 错误信息: {response.text}"

def main():
    print("与模型对话开始，输入'退出'结束对话。")
    while True:
        user_input = input("您: ")
        if user_input.lower() == "退出":
            print("对话结束。")
            break
        response = chat_with_model(user_input)
        print(response)

if __name__ == "__main__":
    main()
